import unittest
from HTMLTestRunner import HTMLTestRunner
import time
import smtplib     #发送邮件模块
from email.mime.text import MIMEText    #定义邮件内容
from email.header import Header         #定义邮件标题
import os       #用于访问操作系统功能的模块
def latest_report(report_dir):#读取最新生成的测试报告
    # os.listdir() 方法用于返回指定的文件夹包含的文件或文件夹的名字的列表
    lists = os.listdir(report_dir)
    # 按时间顺序对该目录文件夹下面的文件进行排序
    lists.sort(key=lambda fn: os.path.getatime(report_dir + '\\' + fn))
    print(("new report is :" + lists[-1]))
    # 输出最新报告的路径
    file = os.path.join(report_dir, lists[-1])
    print(file)
    return file

# 这里所有的参数使用的是163邮箱，如果使用别的邮箱需要自己配置参数
def send_mail(latest_report):  # 邮件发送方法
    f = open(latest_report, 'rb')
    mail_content = f.read()
    f.close()
    smtpserver = 'smtp.qq.com'
    # 发送邮箱用户名密码
    user = '464394562@qq.com'  # 填入自己的邮箱账号
    password = 'rrfraznkuoalbifi'  # 填入自己的授权码，注意非邮件密码
    # 发送和接收邮箱
    sender = '464394562@qq.com'  # 填入发送邮箱的账号
    receives = '2357734471@qq.com'  # 填入接收邮箱的账号
    # 发送邮件主题和内容
    subject = 'Selenium 自动化测试报告'
    # HTML邮件正文
    msg = MIMEText(mail_content, 'html', 'utf-8')
    msg['Subject'] = Header(subject, 'utf-8')
    msg['From'] = sender
    msg['To'] = receives
    smtp = smtplib.SMTP_SSL(smtpserver, 465)
    # HELO 向服务器标识用户身份
    smtp.helo(smtpserver)
    # 服务器返回结果确认
    smtp.ehlo(smtpserver)
    # 登录邮箱服务器用户名和密码
    smtp.login(user, password)
    print("Start send Email...")
    smtp.sendmail(sender, receives, msg.as_string())
    smtp.quit()
    print("Send Email end!")
if __name__ == '__main__':
    # 测试报告存放位置
    report_dir = './test_report'
    # 用例存放位置
    test_dir = './test_report'
    discover = unittest.defaultTestLoader.discover(test_dir, pattern="test*.py")
    now = time.strftime("%Y-%m-%d %H_%M_%S")
    report_name = report_dir + '/' + now + 'result.html'
    with open(report_name, 'wb') as f:
        runner = HTMLTestRunner(stream=f, title="Test Report", description="ERP Test")
        runner.run(discover)
    f.close()
    # 获取最新测试报告
    latest_report = latest_report(report_dir)
    # 发送邮件报告
    send_mail(latest_report)
